from php4dvd.pages.internal_page import InternalPage


class AddMoviePage(InternalPage):

    @property
    def add_button(self):
        return self.driver.find_element_by_css_selector("img[alt=\"Add movie\"]")

    def imdbid_field(self):
        return self.driver.find_element_by_name("imdbid")

    def name_field(self):
        return self.driver.find_element_by_name("name")

    def aka_field(self):
        return self.driver.find_element_by_name("aka")

    def duration_field(self):
        return self.driver.find_element_by_name("duration")

    def rating_field(self):
        return self.driver.find_element_by_name("rating")

    def own_no_radiobutton(self):
        return self.driver.find_element_by_id("own_no")

    def seen_no_radiobutton(self):
        return self.driver.find_element_by_id("seen_no")

    def country_field(self):
        return self.driver.find_element_by_name("country")

    def director_field(self):
        return self.driver.find_element_by_name("director")

    def writer_field(self):
        return self.driver.find_element_by_name("writer")

    def year_field(self):
        return self.driver.find_element_by_name("year")

    def submit_button(self):
        return self.driver.find_element_by_id("submit")